from keras.datasets import mnist
from keras.models import Sequential
from keras.layers.core import Dense, Dropout, Activation, Flatten, Reshape, Layer
from keras.layers.convolutional import Convolution2D, MaxPooling2D, UpSampling2D

def get_model():
    model = Sequential()
    model.add(Dense(500,input_shape=(1138,),activation='sigmoid'))
    model.add(Dense(1138,activation='linear'))
    return model

def get_classify_model():
    model = Sequential()
    model.add(Dense(500,input_shape=(500,),activation='sigmoid'))
    model.add(Dropout(0.25))
    model.add(Dense(2,activation='softmax'))
    return model